FROM registry.openanolis.cn/openanolis/cuda:12.1.1-23-runtime

LABEL \
    maintainer="OpenAnolis AI SIG" \
    org.opencontainers.image.title="tensorflow" \
    org.opencontainers.image.version="2.12.0-23" \
    org.opencontainers.image.date="20231127"

RUN dnf install cuda-compiler-12-1 cuda-command-line-tools-12-1 cuda-cudart-devel-12-1 libcudnn -y
RUN dnf install tensorflow2 -y

CMD ["/bin/bash"]
